// 定义爹找儿子方法：
// 方法三要素：1.参数   2.逻辑   3.返回值

// arr 表示原始数据
// id  表示想要找哪个节点就把哪个节点的id传进去
const findSon = (arr, id = '') => {
  const resultArr = []
  // 逻辑部门
  arr.forEach(item => {
    // 当前传入的id和遍历数组中某一个pid一致，说明当前的item就是传入的id这一项的儿子
    if (item.pid === id) {
      // 找到所有的子集，放到结果中
      resultArr.push(item)
      // findSon就是找儿子的方法
      // 返回的就是儿子的集合 =》 children
      // 将整个children放到他的父亲中
      const children = findSon(arr, item.id)
      // 判断有没有当前项的children字段有没有元素
      if (children.length) {
        item.children = children
      }
    }
  })
  console.log(resultArr)
  return resultArr
}
export default findSon
